package com.example.demo.component;

import org.springframework.stereotype.Component;
import org.springframework.web.socket.WebSocketSession;

import java.util.concurrent.ConcurrentHashMap;

//通过这个类记录当前用户在线的状态,维护userId和WebSocketSession之间的映射
@Component
public class OnlineUserManger {
    private ConcurrentHashMap<Integer, WebSocketSession> sessions = new ConcurrentHashMap<>();

    //1) 用户上限,给这个哈希表里插入键值对
    public void online(int userId,WebSocketSession session){
        if (sessions.get(userId) != null){
            //针对多开的处理
            //此时说明用户已经在线了,就登陆失败,不会记录这个映射关系.
            //不记录这个映射关系,后续就收不到任何消息!
            System.out.println(userId + "已经被登录了,登录失败!");
            return;
        }
        sessions.put(userId,session);
    }

    //2)用户下线,针对这个哈希表删除元素
    public void offline(int userId,WebSocketSession session){
        WebSocketSession existSession = sessions.get(userId);
        if (existSession == session){
            //如果这两个session是同一个,才真正进行下线,否则什么都不做
            sessions.remove(userId);
        }
    }

    //3)根据userId获取到WebSocketSession
    public WebSocketSession getSession(int userId){
        return sessions.get(userId);
    }
}
